# utils.py
import os
from dotenv import load_dotenv
from langchain_openai import ChatOpenAI

load_dotenv()

def get_deepseek_model(model_name="deepseek-chat"):
    return ChatOpenAI(
        model=model_name,
        base_url="https://api.deepseek.com/v1",  # DeepSeek API 地址
        api_key=os.getenv("DEEPSEEK_API_KEY"),
        temperature=0.3,
    )
